ResNet50custom

object ResNet50custom : ONNXModels.CV

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

This model has 50 layers with ResNetv1 architecture.

The model have

an input with the shape (1x224x224x3)
an output with the shape (1x1000)

NOTE: This model is converted from Keras.applications and could be used to be compared with the ResNet50noTopCustom model.

See also

Official ResNet model from Keras.applications.

Functions

model

jvm

open fun model(modelHub: ModelHub): OnnxInferenceModel

pretrainedModel

jvm

open override fun pretrainedModel(modelHub: ModelHub): ImageRecognitionModel

Properties

inputShape

jvm

open val inputShape: LongArray?

Shape of the input accepted by this model, without batch size.

modelRelativePath

jvm

open override val modelRelativePath: String

preprocessor

jvm

open override val preprocessor: Operation<Pair<FloatArray, TensorShape>, Pair<FloatArray, TensorShape>>